package com.aspire.aimini.service;

import com.aspire.aimini.model.entity.AiConfig;
import com.baomidou.mybatisplus.core.metadata.IPage;

import java.util.List;
import java.util.Map;

/**
 * AI配置Service接口
 *
 * @author wangguiyou
 * @date 2024-01-01
 * @description AI配置业务逻辑接口
 */
public interface AiConfigService {

    /**
     * 分页查询AI配置列表
     *
     * @param userId      用户ID
     * @param serviceType 服务类型
     * @param provider    提供商
     * @param status      状态
     * @param page        页码
     * @param size        每页大小
     * @return 分页结果
     */
        IPage<AiConfig> getConfigPage(String serviceType, String provider, Integer status, Integer page,
            Integer size);

    /**
     * 根据ID获取AI配置
     *
     * @param id     配置ID
     * @param userId 用户ID
     * @return AI配置
     */
        AiConfig getConfigById(Long id);

    /**
     * 创建AI配置
     *
     * @param serviceType  服务类型
     * @param provider     提供商
     * @param modelName    模型名称
     * @param apiKey       API密钥
     * @param apiUrl       API地址
     * @param configParams 配置参数
     * @param rateLimit    调用频率限制
     * @param dailyQuota   每日配额
     * @param isDefault    是否默认配置
     * @param userId       用户ID
     * @return 创建的配置
     */
        AiConfig createConfig(String serviceType, String provider, String modelName, String apiKey,
            String apiUrl, String configParams, Integer rateLimit, Integer dailyQuota,
            Boolean isDefault);

    /**
     * 更新AI配置
     *
     * @param id           配置ID
     * @param serviceType  服务类型
     * @param provider     提供商
     * @param modelName    模型名称
     * @param apiKey       API密钥
     * @param apiUrl       API地址
     * @param configParams 配置参数
     * @param rateLimit    调用频率限制
     * @param dailyQuota   每日配额
     * @param isDefault    是否默认配置
     * @param status       状态
     * @param userId       用户ID
     * @return 更新后的配置
     */
        AiConfig updateConfig(Long id, String serviceType, String provider, String modelName, String apiKey,
            String apiUrl, String configParams, Integer rateLimit, Integer dailyQuota,
            Boolean isDefault, Integer status);

    /**
     * 删除AI配置
     *
     * @param id     配置ID
     * @param userId 用户ID
     */
        void deleteConfig(Long id);

    /**
     * 批量删除AI配置
     *
     * @param ids    配置ID列表
     * @param userId 用户ID
     */
        void batchDeleteConfigs(List<Long> ids);

    /**
     * 根据用户ID和服务类型获取配置列表
     *
     * @param userId      用户ID
     * @param serviceType 服务类型
     * @return 配置列表
     */
        List<AiConfig> getConfigsByUserIdAndServiceType(String serviceType);

    /**
     * 获取用户的默认配置
     *
     * @param userId      用户ID
     * @param serviceType 服务类型
     * @return 默认配置
     */
        AiConfig getDefaultConfig(String serviceType);

    /**
     * 设置默认配置
     *
     * @param configId 配置ID
     * @param userId   用户ID
     */
        void setDefaultConfig(Long configId, Long userId);

    /**
     * 测试配置连接
     *
     * @param id     配置ID
     * @param userId 用户ID
     * @return 测试结果
     */
        Map<String, Object> testConfig(Long id);

    /**
     * 增加今日使用量
     *
     * @param id     配置ID
     * @param amount 使用量
     */
    void incrementTodayUsed(Long id, Integer amount);

    /**
     * 检查配置是否可用
     *
     * @param id 配置ID
     * @return 是否可用
     */
    boolean isConfigAvailable(Long id);

    /**
     * 获取系统默认配置
     *
     * @param serviceType 服务类型
     * @return 系统默认配置
     */
    AiConfig getSystemDefaultConfig(String serviceType);

    /**
     * 重置今日使用量
     *
     * @param id 配置ID
     */
    void resetTodayUsed(Long id);

    /**
     * 批量重置今日使用量
     */
    void batchResetTodayUsed();

    /**
     * 根据提供商获取配置列表
     *
     * @param provider 提供商
     * @param status   状态
     * @return 配置列表
     */
    List<AiConfig> getConfigsByProvider(String provider, Integer status);

    /**
     * 统计用户配置数量
     *
     * @param userId 用户ID
     * @return 配置数量
     */
        Long countUserConfigs();

    /**
     * 按服务类型统计配置数量
     *
     * @param userId 用户ID
     * @return 统计结果
     */
        List<Map<String, Object>> countByServiceType();

    /**
     * 按提供商统计配置数量
     *
     * @return 统计结果
     */
    List<Map<String, Object>> countByProvider();

    /**
     * 获取即将达到配额限制的配置
     *
     * @param threshold 阈值(百分比)
     * @return 配置列表
     */
    List<AiConfig> getNearQuotaLimitConfigs(Double threshold);

    /**
     * 获取已超出配额的配置
     *
     * @return 配置列表
     */
    List<AiConfig> getOverQuotaConfigs();

    /**
     * 获取用户配置统计信息
     *
     * @param userId 用户ID
     * @return 统计信息
     */
    Map<String, Object> getUserConfigStats(Long userId);

    /**
     * 复制配置
     *
     * @param id     配置ID
     * @param userId 用户ID
     * @return 复制的配置
     */
    AiConfig copyConfig(Long id, Long userId);

    /**
     * 启用配置
     *
     * @param id     配置ID
     * @param userId 用户ID
     */
    void enableConfig(Long id, Long userId);

    /**
     * 禁用配置
     *
     * @param id     配置ID
     * @param userId 用户ID
     */
    void disableConfig(Long id, Long userId);

    /**
     * 批量更新配置状态
     *
     * @param ids    配置ID列表
     * @param status 状态
     * @param userId 用户ID
     */
    void batchUpdateStatus(List<Long> ids, Integer status, Long userId);

    /**
     * 导入配置
     *
     * @param configData 配置数据
     * @param userId     用户ID
     * @return 导入的配置列表
     */
    List<AiConfig> importConfigs(String configData, Long userId);

    /**
     * 导出配置
     *
     * @param ids    配置ID列表
     * @param userId 用户ID
     * @return 导出的配置数据
     */
    String exportConfigs(List<Long> ids, Long userId);
}